<?xml version="1.0" encoding="ascii"?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
          "DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
<head>
  <title>dadi.RunInParallel</title>
  <link rel="stylesheet" href="epydoc.css" type="text/css" />
  <script type="text/javascript" src="epydoc.js"></script>
</head>

<body bgcolor="white" text="black" link="blue" vlink="#204080"
      alink="#204080">
<!-- ==================== NAVIGATION BAR ==================== -->
<table class="navbar" border="0" width="100%" cellpadding="0"
       bgcolor="#a0c0ff" cellspacing="0">
  <tr valign="middle">

  <!-- Tree link -->
      <th>&nbsp;&nbsp;&nbsp;<a
        href="module-tree.html">Trees</a>&nbsp;&nbsp;&nbsp;</th>

  <!-- Index link -->
      <th>&nbsp;&nbsp;&nbsp;<a
        href="identifier-index.html">Indices</a>&nbsp;&nbsp;&nbsp;</th>

  <!-- Help link -->
      <th>&nbsp;&nbsp;&nbsp;<a
        href="help.html">Help</a>&nbsp;&nbsp;&nbsp;</th>

  <!-- Project homepage -->
      <th class="navbar" align="right" width="100%">
        <table border="0" cellpadding="0" cellspacing="0">
          <tr><th class="navbar" align="center"
            ><a class="navbar" target="_top" href="http://dadi.googlecode.com">dadi</a></th>
          </tr></table></th>
  </tr>
</table>
<table width="100%" cellpadding="0" cellspacing="0">
  <tr valign="top">
    <td width="100%">
      <span class="breadcrumbs">
        <a href="dadi-module.html">Package&nbsp;dadi</a> ::
        Module&nbsp;RunInParallel
      </span>
    </td>
    <td>
      <table cellpadding="0" cellspacing="0">
        <!-- hide/show private -->
        <tr><td align="right"><span class="options">[<a href="javascript:void(0);" class="privatelink"
    onclick="toggle_private();">hide&nbsp;private</a>]</span></td></tr>
        <tr><td align="right"><span class="options"
            >[<a href="frames.html" target="_top">frames</a
            >]&nbsp;|&nbsp;<a href="dadi.RunInParallel-pysrc.html"
            target="_top">no&nbsp;frames</a>]</span></td></tr>
      </table>
    </td>
  </tr>
</table>
<h1 class="epydoc">Source Code for <a href="dadi.RunInParallel-module.html">Module dadi.RunInParallel</a></h1>
<pre class="py-src">
<a name="L1"></a><tt class="py-lineno">  1</tt>  <tt class="py-line"><tt class="py-docstring">"""</tt> </tt>
<a name="L2"></a><tt class="py-lineno">  2</tt>  <tt class="py-line"><tt class="py-docstring">For evaluation of gradients by parallel finite-difference calculation.</tt> </tt>
<a name="L3"></a><tt class="py-lineno">  3</tt>  <tt class="py-line"><tt class="py-docstring"></tt> </tt>
<a name="L4"></a><tt class="py-lineno">  4</tt>  <tt class="py-line"><tt class="py-docstring">This code has not been tested thoroughly.</tt> </tt>
<a name="L5"></a><tt class="py-lineno">  5</tt>  <tt class="py-line"><tt class="py-docstring">"""</tt> </tt>
<a name="L6"></a><tt class="py-lineno">  6</tt>  <tt class="py-line"><tt class="py-keyword">import</tt> <tt class="py-name">logging</tt> </tt>
<a name="L7"></a><tt class="py-lineno">  7</tt>  <tt class="py-line"><tt class="py-name">logging</tt><tt class="py-op">.</tt><tt class="py-name">basicConfig</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L8"></a><tt class="py-lineno">  8</tt>  <tt class="py-line"><tt id="link-0" class="py-name" targets="Variable dadi.Hessian.logger=dadi.Hessian-module.html#logger,Variable dadi.Inference.logger=dadi.Inference-module.html#logger,Variable dadi.Integration.logger=dadi.Integration-module.html#logger,Variable dadi.Numerics.logger=dadi.Numerics-module.html#logger,Variable dadi.RunInParallel.logger=dadi.RunInParallel-module.html#logger,Variable dadi.Spectrum_mod.logger=dadi.Spectrum_mod-module.html#logger"><a title="dadi.Hessian.logger
dadi.Inference.logger
dadi.Integration.logger
dadi.Numerics.logger
dadi.RunInParallel.logger
dadi.Spectrum_mod.logger" class="py-name" href="#" onclick="return doclink('link-0', 'logger', 'link-0');">logger</a></tt> <tt class="py-op">=</tt> <tt class="py-name">logging</tt><tt class="py-op">.</tt><tt class="py-name">getLogger</tt><tt class="py-op">(</tt><tt class="py-string">'RunInParallel'</tt><tt class="py-op">)</tt> </tt>
<a name="L9"></a><tt class="py-lineno">  9</tt>  <tt class="py-line"> </tt>
<a name="L10"></a><tt class="py-lineno"> 10</tt>  <tt class="py-line"><tt class="py-keyword">import</tt> <tt class="py-name">cPickle</tt><tt class="py-op">,</tt> <tt class="py-name">os</tt><tt class="py-op">,</tt> <tt class="py-name">sys</tt><tt class="py-op">,</tt> <tt class="py-name">traceback</tt> </tt>
<a name="L11"></a><tt class="py-lineno"> 11</tt>  <tt class="py-line"> </tt>
<a name="L12"></a><tt class="py-lineno"> 12</tt>  <tt class="py-line"><tt class="py-comment"># With some versions of pypar and on some systems, importing pypar changes</tt> </tt>
<a name="L13"></a><tt class="py-lineno"> 13</tt>  <tt class="py-line"><tt class="py-comment"># the current working directory. We save it here so we can change back.</tt> </tt>
<a name="L14"></a><tt class="py-lineno"> 14</tt>  <tt class="py-line"><tt id="link-1" class="py-name" targets="Variable dadi.RunInParallel.currdir=dadi.RunInParallel-module.html#currdir"><a title="dadi.RunInParallel.currdir" class="py-name" href="#" onclick="return doclink('link-1', 'currdir', 'link-1');">currdir</a></tt> <tt class="py-op">=</tt> <tt class="py-name">os</tt><tt class="py-op">.</tt><tt class="py-name">getcwd</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L15"></a><tt class="py-lineno"> 15</tt>  <tt class="py-line"><tt class="py-keyword">try</tt><tt class="py-op">:</tt> </tt>
<a name="L16"></a><tt class="py-lineno"> 16</tt>  <tt class="py-line">    <tt class="py-keyword">import</tt> <tt class="py-name">pypar</tt> </tt>
<a name="L17"></a><tt class="py-lineno"> 17</tt>  <tt class="py-line">    <tt id="link-2" class="py-name" targets="Variable dadi.RunInParallel.HAVE_PYPAR=dadi.RunInParallel-module.html#HAVE_PYPAR"><a title="dadi.RunInParallel.HAVE_PYPAR" class="py-name" href="#" onclick="return doclink('link-2', 'HAVE_PYPAR', 'link-2');">HAVE_PYPAR</a></tt> <tt class="py-op">=</tt> <tt class="py-name">True</tt> </tt>
<a name="L18"></a><tt class="py-lineno"> 18</tt>  <tt class="py-line">    <tt id="link-3" class="py-name" targets="Variable dadi.RunInParallel.num_procs=dadi.RunInParallel-module.html#num_procs"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-3', 'num_procs', 'link-3');">num_procs</a></tt> <tt class="py-op">=</tt> <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">size</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L19"></a><tt class="py-lineno"> 19</tt>  <tt class="py-line">    <tt id="link-4" class="py-name" targets="Variable dadi.RunInParallel.my_rank=dadi.RunInParallel-module.html#my_rank"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-4', 'my_rank', 'link-4');">my_rank</a></tt> <tt class="py-op">=</tt> <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">rank</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L20"></a><tt class="py-lineno"> 20</tt>  <tt class="py-line">    <tt id="link-5" class="py-name" targets="Variable dadi.RunInParallel.my_host=dadi.RunInParallel-module.html#my_host"><a title="dadi.RunInParallel.my_host" class="py-name" href="#" onclick="return doclink('link-5', 'my_host', 'link-5');">my_host</a></tt> <tt class="py-op">=</tt> <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">get_processor_name</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L21"></a><tt class="py-lineno"> 21</tt>  <tt class="py-line">    <tt class="py-keyword">import</tt> <tt class="py-name">atexit</tt> </tt>
<a name="L22"></a><tt class="py-lineno"> 22</tt>  <tt class="py-line">    <tt class="py-name">atexit</tt><tt class="py-op">.</tt><tt class="py-name">register</tt><tt class="py-op">(</tt><tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">finalize</tt><tt class="py-op">)</tt> </tt>
<a name="L23"></a><tt class="py-lineno"> 23</tt>  <tt class="py-line"><tt class="py-keyword">except</tt> <tt class="py-name">ImportError</tt><tt class="py-op">:</tt> </tt>
<a name="L24"></a><tt class="py-lineno"> 24</tt>  <tt class="py-line">    <tt id="link-6" class="py-name"><a title="dadi.RunInParallel.HAVE_PYPAR" class="py-name" href="#" onclick="return doclink('link-6', 'HAVE_PYPAR', 'link-2');">HAVE_PYPAR</a></tt> <tt class="py-op">=</tt> <tt class="py-name">False</tt> </tt>
<a name="L25"></a><tt class="py-lineno"> 25</tt>  <tt class="py-line">    <tt id="link-7" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-7', 'num_procs', 'link-3');">num_procs</a></tt> <tt class="py-op">=</tt> <tt class="py-number">1</tt> </tt>
<a name="L26"></a><tt class="py-lineno"> 26</tt>  <tt class="py-line">    <tt id="link-8" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-8', 'my_rank', 'link-4');">my_rank</a></tt> <tt class="py-op">=</tt> <tt class="py-number">0</tt> </tt>
<a name="L27"></a><tt class="py-lineno"> 27</tt>  <tt class="py-line">    <tt class="py-keyword">import</tt> <tt class="py-name">socket</tt> </tt>
<a name="L28"></a><tt class="py-lineno"> 28</tt>  <tt class="py-line">    <tt id="link-9" class="py-name"><a title="dadi.RunInParallel.my_host" class="py-name" href="#" onclick="return doclink('link-9', 'my_host', 'link-5');">my_host</a></tt> <tt class="py-op">=</tt> <tt class="py-name">socket</tt><tt class="py-op">.</tt><tt class="py-name">gethostname</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L29"></a><tt class="py-lineno"> 29</tt>  <tt class="py-line"><tt class="py-name">os</tt><tt class="py-op">.</tt><tt class="py-name">chdir</tt><tt class="py-op">(</tt><tt id="link-10" class="py-name"><a title="dadi.RunInParallel.currdir" class="py-name" href="#" onclick="return doclink('link-10', 'currdir', 'link-1');">currdir</a></tt><tt class="py-op">)</tt> </tt>
<a name="L30"></a><tt class="py-lineno"> 30</tt>  <tt class="py-line"><tt id="link-11" class="py-name"><a title="dadi.Hessian.logger
dadi.Inference.logger
dadi.Integration.logger
dadi.Numerics.logger
dadi.RunInParallel.logger
dadi.Spectrum_mod.logger" class="py-name" href="#" onclick="return doclink('link-11', 'logger', 'link-0');">logger</a></tt><tt class="py-op">.</tt><tt class="py-name">debug</tt><tt class="py-op">(</tt><tt class="py-string">'Node %i is on host %s.'</tt> <tt class="py-op">%</tt> <tt class="py-op">(</tt><tt id="link-12" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-12', 'my_rank', 'link-4');">my_rank</a></tt><tt class="py-op">,</tt> <tt id="link-13" class="py-name"><a title="dadi.RunInParallel.my_host" class="py-name" href="#" onclick="return doclink('link-13', 'my_host', 'link-5');">my_host</a></tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
<a name="L31"></a><tt class="py-lineno"> 31</tt>  <tt class="py-line"> </tt>
<a name="Statement"></a><div id="Statement-def"><a name="L32"></a><tt class="py-lineno"> 32</tt> <a class="py-toggle" href="#" id="Statement-toggle" onclick="return toggle('Statement');">-</a><tt class="py-line"><tt class="py-keyword">class</tt> <a class="py-def-name" href="dadi.RunInParallel.Statement-class.html">Statement</a><tt class="py-op">:</tt> </tt>
</div><div id="Statement-collapsed" style="display:none;" pad="+++" indent="++++"></div><div id="Statement-expanded"><a name="L33"></a><tt class="py-lineno"> 33</tt>  <tt class="py-line">    <tt class="py-docstring">"""</tt> </tt>
<a name="L34"></a><tt class="py-lineno"> 34</tt>  <tt class="py-line"><tt class="py-docstring">    Class for sending Python statements to workers.</tt> </tt>
<a name="L35"></a><tt class="py-lineno"> 35</tt>  <tt class="py-line"><tt class="py-docstring">    """</tt> </tt>
<a name="Statement.__init__"></a><div id="Statement.__init__-def"><a name="L36"></a><tt class="py-lineno"> 36</tt> <a class="py-toggle" href="#" id="Statement.__init__-toggle" onclick="return toggle('Statement.__init__');">-</a><tt class="py-line">    <tt class="py-keyword">def</tt> <a class="py-def-name" href="dadi.RunInParallel.Statement-class.html#__init__">__init__</a><tt class="py-op">(</tt><tt class="py-param">self</tt><tt class="py-op">,</tt> <tt class="py-param">statement</tt><tt class="py-op">,</tt> <tt class="py-param">locals</tt><tt class="py-op">=</tt><tt class="py-op">{</tt><tt class="py-op">}</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
</div><div id="Statement.__init__-collapsed" style="display:none;" pad="+++" indent="++++++++"></div><div id="Statement.__init__-expanded"><a name="L37"></a><tt class="py-lineno"> 37</tt>  <tt class="py-line">        <tt class="py-name">self</tt><tt class="py-op">.</tt><tt class="py-name">statement</tt> <tt class="py-op">=</tt> <tt class="py-name">statement</tt> </tt>
<a name="L38"></a><tt class="py-lineno"> 38</tt>  <tt class="py-line">        <tt class="py-name">self</tt><tt class="py-op">.</tt><tt class="py-name">locals</tt> <tt class="py-op">=</tt> <tt class="py-name">locals</tt> </tt>
</div></div><a name="L39"></a><tt class="py-lineno"> 39</tt>  <tt class="py-line"> </tt>
<a name="eval_func_over_list"></a><div id="eval_func_over_list-def"><a name="L40"></a><tt class="py-lineno"> 40</tt> <a class="py-toggle" href="#" id="eval_func_over_list-toggle" onclick="return toggle('eval_func_over_list');">-</a><tt class="py-line"><tt class="py-keyword">def</tt> <a class="py-def-name" href="dadi.RunInParallel-module.html#eval_func_over_list">eval_func_over_list</a><tt class="py-op">(</tt><tt class="py-param">func_str</tt><tt class="py-op">,</tt> <tt class="py-param">x_todo</tt><tt class="py-op">,</tt> <tt class="py-op">*</tt><tt class="py-param">args</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
</div><div id="eval_func_over_list-collapsed" style="display:none;" pad="+++" indent="++++"></div><div id="eval_func_over_list-expanded"><a name="L41"></a><tt class="py-lineno"> 41</tt>  <tt class="py-line">    <tt class="py-keyword">return</tt> <tt class="py-op">[</tt><tt class="py-name">eval</tt><tt class="py-op">(</tt><tt class="py-string">'%s(x, *args)'</tt> <tt class="py-op">%</tt> <tt class="py-name">func_str</tt><tt class="py-op">)</tt> <tt class="py-keyword">for</tt> <tt class="py-name">x</tt> <tt class="py-keyword">in</tt> <tt class="py-name">x_todo</tt><tt class="py-op">]</tt> </tt>
</div><a name="L42"></a><tt class="py-lineno"> 42</tt>  <tt class="py-line"> </tt>
<a name="L43"></a><tt class="py-lineno"> 43</tt>  <tt class="py-line"><tt class="py-keyword">while</tt> <tt id="link-14" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-14', 'my_rank', 'link-4');">my_rank</a></tt> <tt class="py-op">!=</tt> <tt class="py-number">0</tt><tt class="py-op">:</tt> </tt>
<a name="L44"></a><tt class="py-lineno"> 44</tt>  <tt class="py-line">    <tt class="py-comment"># Wait for a message</tt> </tt>
<a name="L45"></a><tt class="py-lineno"> 45</tt>  <tt class="py-line">    <tt class="py-name">message</tt> <tt class="py-op">=</tt> <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">receive</tt><tt class="py-op">(</tt><tt class="py-name">source</tt><tt class="py-op">=</tt><tt class="py-number">0</tt><tt class="py-op">)</tt> </tt>
<a name="L46"></a><tt class="py-lineno"> 46</tt>  <tt class="py-line"> </tt>
<a name="L47"></a><tt class="py-lineno"> 47</tt>  <tt class="py-line">    <tt class="py-comment"># If the message is a SystemExit exception, exit the code.</tt> </tt>
<a name="L48"></a><tt class="py-lineno"> 48</tt>  <tt class="py-line">    <tt class="py-keyword">if</tt> <tt class="py-name">isinstance</tt><tt class="py-op">(</tt><tt class="py-name">message</tt><tt class="py-op">,</tt> <tt class="py-name">SystemExit</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L49"></a><tt class="py-lineno"> 49</tt>  <tt class="py-line">        <tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">exit</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L50"></a><tt class="py-lineno"> 50</tt>  <tt class="py-line"> </tt>
<a name="L51"></a><tt class="py-lineno"> 51</tt>  <tt class="py-line">    <tt class="py-comment"># Exception handling:</tt> </tt>
<a name="L52"></a><tt class="py-lineno"> 52</tt>  <tt class="py-line">    <tt class="py-comment">#    If we catch any exception, it's probably a bug in the code. Print</tt> </tt>
<a name="L53"></a><tt class="py-lineno"> 53</tt>  <tt class="py-line">    <tt class="py-comment">#      a nice traceback, save results, and exit the code.</tt> </tt>
<a name="L54"></a><tt class="py-lineno"> 54</tt>  <tt class="py-line">    <tt class="py-keyword">try</tt><tt class="py-op">:</tt> </tt>
<a name="L55"></a><tt class="py-lineno"> 55</tt>  <tt class="py-line">        <tt class="py-keyword">if</tt> <tt class="py-name">isinstance</tt><tt class="py-op">(</tt><tt class="py-name">message</tt><tt class="py-op">,</tt> <tt id="link-15" class="py-name" targets="Class dadi.RunInParallel.Statement=dadi.RunInParallel.Statement-class.html"><a title="dadi.RunInParallel.Statement" class="py-name" href="#" onclick="return doclink('link-15', 'Statement', 'link-15');">Statement</a></tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L56"></a><tt class="py-lineno"> 56</tt>  <tt class="py-line">            <tt class="py-name">command</tt><tt class="py-op">,</tt> <tt class="py-name">msg_locals</tt> <tt class="py-op">=</tt> <tt class="py-name">message</tt><tt class="py-op">.</tt><tt class="py-name">statement</tt><tt class="py-op">,</tt> <tt class="py-name">message</tt><tt class="py-op">.</tt><tt class="py-name">locals</tt> </tt>
<a name="L57"></a><tt class="py-lineno"> 57</tt>  <tt class="py-line">            <tt class="py-name">locals</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">.</tt><tt class="py-name">update</tt><tt class="py-op">(</tt><tt class="py-name">msg_locals</tt><tt class="py-op">)</tt> </tt>
<a name="L58"></a><tt class="py-lineno"> 58</tt>  <tt class="py-line">            <tt class="py-keyword">exec</tt><tt class="py-op">(</tt><tt class="py-name">command</tt><tt class="py-op">)</tt> </tt>
<a name="L59"></a><tt class="py-lineno"> 59</tt>  <tt class="py-line">        <tt class="py-keyword">else</tt><tt class="py-op">:</tt> </tt>
<a name="L60"></a><tt class="py-lineno"> 60</tt>  <tt class="py-line">            <tt class="py-name">command</tt><tt class="py-op">,</tt> <tt class="py-name">msg_locals</tt> <tt class="py-op">=</tt> <tt class="py-name">message</tt>  </tt>
<a name="L61"></a><tt class="py-lineno"> 61</tt>  <tt class="py-line">            <tt class="py-name">locals</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">.</tt><tt class="py-name">update</tt><tt class="py-op">(</tt><tt class="py-name">msg_locals</tt><tt class="py-op">)</tt> </tt>
<a name="L62"></a><tt class="py-lineno"> 62</tt>  <tt class="py-line">            <tt class="py-name">result</tt> <tt class="py-op">=</tt> <tt class="py-name">eval</tt><tt class="py-op">(</tt><tt class="py-name">command</tt><tt class="py-op">)</tt> </tt>
<a name="L63"></a><tt class="py-lineno"> 63</tt>  <tt class="py-line">            <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">send</tt><tt class="py-op">(</tt><tt class="py-name">result</tt><tt class="py-op">,</tt> <tt class="py-number">0</tt><tt class="py-op">)</tt> </tt>
<a name="L64"></a><tt class="py-lineno"> 64</tt>  <tt class="py-line">    <tt class="py-keyword">except</tt><tt class="py-op">:</tt> </tt>
<a name="L65"></a><tt class="py-lineno"> 65</tt>  <tt class="py-line">        <tt class="py-comment"># Assemble and print a nice traceback</tt> </tt>
<a name="L66"></a><tt class="py-lineno"> 66</tt>  <tt class="py-line">        <tt class="py-name">tb</tt> <tt class="py-op">=</tt> <tt class="py-name">traceback</tt><tt class="py-op">.</tt><tt class="py-name">format_exception</tt><tt class="py-op">(</tt><tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">exc_type</tt><tt class="py-op">,</tt> <tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">exc_value</tt><tt class="py-op">,</tt>  </tt>
<a name="L67"></a><tt class="py-lineno"> 67</tt>  <tt class="py-line">                                        <tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">exc_traceback</tt><tt class="py-op">)</tt> </tt>
<a name="L68"></a><tt class="py-lineno"> 68</tt>  <tt class="py-line">        <tt id="link-16" class="py-name"><a title="dadi.Hessian.logger
dadi.Inference.logger
dadi.Integration.logger
dadi.Numerics.logger
dadi.RunInParallel.logger
dadi.Spectrum_mod.logger" class="py-name" href="#" onclick="return doclink('link-16', 'logger', 'link-0');">logger</a></tt><tt class="py-op">.</tt><tt class="py-name">critical</tt><tt class="py-op">(</tt><tt class="py-name">os</tt><tt class="py-op">.</tt><tt class="py-name">getcwd</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
<a name="L69"></a><tt class="py-lineno"> 69</tt>  <tt class="py-line">        <tt id="link-17" class="py-name"><a title="dadi.Hessian.logger
dadi.Inference.logger
dadi.Integration.logger
dadi.Numerics.logger
dadi.RunInParallel.logger
dadi.Spectrum_mod.logger" class="py-name" href="#" onclick="return doclink('link-17', 'logger', 'link-0');">logger</a></tt><tt class="py-op">.</tt><tt class="py-name">critical</tt><tt class="py-op">(</tt><tt class="py-op">(</tt><tt class="py-string">'node %i:'</tt><tt class="py-op">%</tt><tt id="link-18" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-18', 'my_rank', 'link-4');">my_rank</a></tt><tt class="py-op">)</tt><tt class="py-op">.</tt><tt class="py-name">join</tt><tt class="py-op">(</tt><tt class="py-name">tb</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
<a name="L70"></a><tt class="py-lineno"> 70</tt>  <tt class="py-line">        <tt class="py-name">save_to</tt> <tt class="py-op">=</tt> <tt class="py-string">'node_%i_crash.bp'</tt> <tt class="py-op">%</tt> <tt id="link-19" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-19', 'my_rank', 'link-4');">my_rank</a></tt> </tt>
<a name="L71"></a><tt class="py-lineno"> 71</tt>  <tt class="py-line">        <tt id="link-20" class="py-name"><a title="dadi.Hessian.logger
dadi.Inference.logger
dadi.Integration.logger
dadi.Numerics.logger
dadi.RunInParallel.logger
dadi.Spectrum_mod.logger" class="py-name" href="#" onclick="return doclink('link-20', 'logger', 'link-0');">logger</a></tt><tt class="py-op">.</tt><tt class="py-name">critical</tt><tt class="py-op">(</tt><tt class="py-string">"node %i: Command being run was: %s."</tt> </tt>
<a name="L72"></a><tt class="py-lineno"> 72</tt>  <tt class="py-line">                        <tt class="py-op">%</tt> <tt class="py-op">(</tt><tt id="link-21" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-21', 'my_rank', 'link-4');">my_rank</a></tt><tt class="py-op">,</tt> <tt class="py-name">command</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
<a name="L73"></a><tt class="py-lineno"> 73</tt>  <tt class="py-line">        <tt class="py-name">dump_file</tt> <tt class="py-op">=</tt> <tt class="py-name">file</tt><tt class="py-op">(</tt><tt class="py-name">save_to</tt><tt class="py-op">,</tt> <tt class="py-string">'w'</tt><tt class="py-op">)</tt> </tt>
<a name="L74"></a><tt class="py-lineno"> 74</tt>  <tt class="py-line">        <tt class="py-name">cPickle</tt><tt class="py-op">.</tt><tt class="py-name">dump</tt><tt class="py-op">(</tt><tt class="py-name">msg_locals</tt><tt class="py-op">,</tt> <tt class="py-name">dump_file</tt><tt class="py-op">)</tt> </tt>
<a name="L75"></a><tt class="py-lineno"> 75</tt>  <tt class="py-line">        <tt class="py-name">dump_file</tt><tt class="py-op">.</tt><tt class="py-name">close</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L76"></a><tt class="py-lineno"> 76</tt>  <tt class="py-line">        <tt id="link-22" class="py-name"><a title="dadi.Hessian.logger
dadi.Inference.logger
dadi.Integration.logger
dadi.Numerics.logger
dadi.RunInParallel.logger
dadi.Spectrum_mod.logger" class="py-name" href="#" onclick="return doclink('link-22', 'logger', 'link-0');">logger</a></tt><tt class="py-op">.</tt><tt class="py-name">critical</tt><tt class="py-op">(</tt><tt class="py-string">"node %i: Corresponding locals saved to %s."</tt> </tt>
<a name="L77"></a><tt class="py-lineno"> 77</tt>  <tt class="py-line">                        <tt class="py-op">%</tt> <tt class="py-op">(</tt><tt id="link-23" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-23', 'my_rank', 'link-4');">my_rank</a></tt><tt class="py-op">,</tt> <tt class="py-name">save_to</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
<a name="L78"></a><tt class="py-lineno"> 78</tt>  <tt class="py-line">        <tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">exit</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L79"></a><tt class="py-lineno"> 79</tt>  <tt class="py-line"> </tt>
<a name="stop_workers"></a><div id="stop_workers-def"><a name="L80"></a><tt class="py-lineno"> 80</tt> <a class="py-toggle" href="#" id="stop_workers-toggle" onclick="return toggle('stop_workers');">-</a><tt class="py-line"><tt class="py-keyword">def</tt> <a class="py-def-name" href="dadi.RunInParallel-module.html#stop_workers">stop_workers</a><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
</div><div id="stop_workers-collapsed" style="display:none;" pad="+++" indent="++++"></div><div id="stop_workers-expanded"><a name="L81"></a><tt class="py-lineno"> 81</tt>  <tt class="py-line">    <tt class="py-docstring">"""</tt> </tt>
<a name="L82"></a><tt class="py-lineno"> 82</tt>  <tt class="py-line"><tt class="py-docstring">    Send all workers the command to exit the program.</tt> </tt>
<a name="L83"></a><tt class="py-lineno"> 83</tt>  <tt class="py-line"><tt class="py-docstring">    """</tt> </tt>
<a name="L84"></a><tt class="py-lineno"> 84</tt>  <tt class="py-line">    <tt class="py-keyword">for</tt> <tt class="py-name">worker</tt> <tt class="py-keyword">in</tt> <tt class="py-name">range</tt><tt class="py-op">(</tt><tt class="py-number">1</tt><tt class="py-op">,</tt> <tt id="link-24" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-24', 'num_procs', 'link-3');">num_procs</a></tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L85"></a><tt class="py-lineno"> 85</tt>  <tt class="py-line">        <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">send</tt><tt class="py-op">(</tt><tt class="py-name">SystemExit</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">,</tt> <tt class="py-name">worker</tt><tt class="py-op">)</tt> </tt>
</div><a name="L86"></a><tt class="py-lineno"> 86</tt>  <tt class="py-line"> </tt>
<a name="L87"></a><tt class="py-lineno"> 87</tt>  <tt class="py-line"><tt class="py-keyword">if</tt> <tt id="link-25" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-25', 'my_rank', 'link-4');">my_rank</a></tt> <tt class="py-op">==</tt> <tt class="py-number">0</tt><tt class="py-op">:</tt> </tt>
<a name="L88"></a><tt class="py-lineno"> 88</tt>  <tt class="py-line">    <tt class="py-keyword">import</tt> <tt class="py-name">atexit</tt> </tt>
<a name="L89"></a><tt class="py-lineno"> 89</tt>  <tt class="py-line">    <tt class="py-name">atexit</tt><tt class="py-op">.</tt><tt class="py-name">register</tt><tt class="py-op">(</tt><tt id="link-26" class="py-name" targets="Function dadi.RunInParallel.stop_workers()=dadi.RunInParallel-module.html#stop_workers"><a title="dadi.RunInParallel.stop_workers" class="py-name" href="#" onclick="return doclink('link-26', 'stop_workers', 'link-26');">stop_workers</a></tt><tt class="py-op">)</tt> </tt>
<a name="L90"></a><tt class="py-lineno"> 90</tt>  <tt class="py-line"> </tt>
<a name="statement_to_all_workers"></a><div id="statement_to_all_workers-def"><a name="L91"></a><tt class="py-lineno"> 91</tt> <a class="py-toggle" href="#" id="statement_to_all_workers-toggle" onclick="return toggle('statement_to_all_workers');">-</a><tt class="py-line"><tt class="py-keyword">def</tt> <a class="py-def-name" href="dadi.RunInParallel-module.html#statement_to_all_workers">statement_to_all_workers</a><tt class="py-op">(</tt><tt class="py-param">statement</tt><tt class="py-op">,</tt> <tt class="py-param">locals</tt><tt class="py-op">=</tt><tt class="py-op">{</tt><tt class="py-op">}</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
</div><div id="statement_to_all_workers-collapsed" style="display:none;" pad="+++" indent="++++"></div><div id="statement_to_all_workers-expanded"><a name="L92"></a><tt class="py-lineno"> 92</tt>  <tt class="py-line">    <tt class="py-docstring">"""</tt> </tt>
<a name="L93"></a><tt class="py-lineno"> 93</tt>  <tt class="py-line"><tt class="py-docstring">    Send a Python statement to all workers for execution.</tt> </tt>
<a name="L94"></a><tt class="py-lineno"> 94</tt>  <tt class="py-line"><tt class="py-docstring">    """</tt> </tt>
<a name="L95"></a><tt class="py-lineno"> 95</tt>  <tt class="py-line">    <tt class="py-keyword">for</tt> <tt class="py-name">worker</tt> <tt class="py-keyword">in</tt> <tt class="py-name">range</tt><tt class="py-op">(</tt><tt class="py-number">1</tt><tt class="py-op">,</tt> <tt id="link-27" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-27', 'num_procs', 'link-3');">num_procs</a></tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L96"></a><tt class="py-lineno"> 96</tt>  <tt class="py-line">        <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">send</tt><tt class="py-op">(</tt><tt id="link-28" class="py-name"><a title="dadi.RunInParallel.Statement" class="py-name" href="#" onclick="return doclink('link-28', 'Statement', 'link-15');">Statement</a></tt><tt class="py-op">(</tt><tt class="py-name">statement</tt><tt class="py-op">,</tt> <tt class="py-name">locals</tt><tt class="py-op">)</tt><tt class="py-op">,</tt> <tt class="py-name">worker</tt><tt class="py-op">)</tt> </tt>
</div><a name="L97"></a><tt class="py-lineno"> 97</tt>  <tt class="py-line"> </tt>
<a name="L98"></a><tt class="py-lineno"> 98</tt>  <tt class="py-line"><tt class="py-comment"># Ensure that all workers are working in the correct directory.</tt> </tt>
<a name="L99"></a><tt class="py-lineno"> 99</tt>  <tt class="py-line"><tt class="py-keyword">if</tt> <tt id="link-29" class="py-name"><a title="dadi.RunInParallel.my_rank" class="py-name" href="#" onclick="return doclink('link-29', 'my_rank', 'link-4');">my_rank</a></tt> <tt class="py-op">==</tt> <tt class="py-number">0</tt><tt class="py-op">:</tt> </tt>
<a name="L100"></a><tt class="py-lineno">100</tt>  <tt class="py-line">    <tt id="link-30" class="py-name" targets="Function dadi.RunInParallel.statement_to_all_workers()=dadi.RunInParallel-module.html#statement_to_all_workers"><a title="dadi.RunInParallel.statement_to_all_workers" class="py-name" href="#" onclick="return doclink('link-30', 'statement_to_all_workers', 'link-30');">statement_to_all_workers</a></tt><tt class="py-op">(</tt><tt class="py-string">"os.chdir('%s')"</tt> <tt class="py-op">%</tt> <tt id="link-31" class="py-name"><a title="dadi.RunInParallel.currdir" class="py-name" href="#" onclick="return doclink('link-31', 'currdir', 'link-1');">currdir</a></tt><tt class="py-op">)</tt> </tt>
<a name="L101"></a><tt class="py-lineno">101</tt>  <tt class="py-line">    <tt id="link-32" class="py-name"><a title="dadi.RunInParallel.statement_to_all_workers" class="py-name" href="#" onclick="return doclink('link-32', 'statement_to_all_workers', 'link-30');">statement_to_all_workers</a></tt><tt class="py-op">(</tt><tt class="py-string">"sys.path = master_path"</tt><tt class="py-op">,</tt> <tt class="py-op">{</tt><tt class="py-string">'master_path'</tt><tt class="py-op">:</tt><tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">path</tt><tt class="py-op">}</tt><tt class="py-op">)</tt> </tt>
<a name="L102"></a><tt class="py-lineno">102</tt>  <tt class="py-line"> </tt>
<a name="L103"></a><tt class="py-lineno">103</tt>  <tt class="py-line"><tt class="py-keyword">import</tt> <tt class="py-name">numpy</tt> </tt>
<a name="make_parallel_gradient_func"></a><div id="make_parallel_gradient_func-def"><a name="L104"></a><tt class="py-lineno">104</tt> <a class="py-toggle" href="#" id="make_parallel_gradient_func-toggle" onclick="return toggle('make_parallel_gradient_func');">-</a><tt class="py-line"><tt class="py-keyword">def</tt> <a class="py-def-name" href="dadi.RunInParallel-module.html#make_parallel_gradient_func">make_parallel_gradient_func</a><tt class="py-op">(</tt><tt class="py-param">func_str</tt><tt class="py-op">,</tt> <tt class="py-param">epsilon</tt><tt class="py-op">=</tt><tt class="py-number">1e-8</tt><tt class="py-op">,</tt> <tt class="py-param">send_grad_args</tt><tt class="py-op">=</tt><tt class="py-name">True</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
</div><div id="make_parallel_gradient_func-collapsed" style="display:none;" pad="+++" indent="++++"></div><div id="make_parallel_gradient_func-expanded"><a name="L105"></a><tt class="py-lineno">105</tt>  <tt class="py-line">    <tt class="py-keyword">def</tt> <tt class="py-def-name">grad_func</tt><tt class="py-op">(</tt><tt class="py-param">x0</tt><tt class="py-op">,</tt> <tt class="py-op">*</tt><tt class="py-param">grad_args</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L106"></a><tt class="py-lineno">106</tt>  <tt class="py-line">        <tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">stdout</tt><tt class="py-op">.</tt><tt class="py-name">flush</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L107"></a><tt class="py-lineno">107</tt>  <tt class="py-line">        <tt class="py-name">x0</tt> <tt class="py-op">=</tt> <tt class="py-name">numpy</tt><tt class="py-op">.</tt><tt class="py-name">asarray</tt><tt class="py-op">(</tt><tt class="py-name">x0</tt><tt class="py-op">)</tt> </tt>
<a name="L108"></a><tt class="py-lineno">108</tt>  <tt class="py-line"> </tt>
<a name="L109"></a><tt class="py-lineno">109</tt>  <tt class="py-line">        <tt class="py-comment"># Generate a list of parameter sets to evaluate the function at</tt> </tt>
<a name="L110"></a><tt class="py-lineno">110</tt>  <tt class="py-line">        <tt class="py-name">x_todo</tt> <tt class="py-op">=</tt> <tt class="py-op">[</tt><tt class="py-name">x0</tt><tt class="py-op">]</tt> </tt>
<a name="L111"></a><tt class="py-lineno">111</tt>  <tt class="py-line">        <tt class="py-keyword">for</tt> <tt class="py-name">ii</tt> <tt class="py-keyword">in</tt> <tt class="py-name">range</tt><tt class="py-op">(</tt><tt class="py-name">len</tt><tt class="py-op">(</tt><tt class="py-name">x0</tt><tt class="py-op">)</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L112"></a><tt class="py-lineno">112</tt>  <tt class="py-line">            <tt class="py-name">eps</tt> <tt class="py-op">=</tt> <tt class="py-name">numpy</tt><tt class="py-op">.</tt><tt class="py-name">zeros</tt><tt class="py-op">(</tt><tt class="py-name">len</tt><tt class="py-op">(</tt><tt class="py-name">x0</tt><tt class="py-op">)</tt><tt class="py-op">,</tt> <tt class="py-name">numpy</tt><tt class="py-op">.</tt><tt class="py-name">float_</tt><tt class="py-op">)</tt> </tt>
<a name="L113"></a><tt class="py-lineno">113</tt>  <tt class="py-line">            <tt class="py-name">eps</tt><tt class="py-op">[</tt><tt class="py-name">ii</tt><tt class="py-op">]</tt> <tt class="py-op">=</tt> <tt class="py-name">epsilon</tt> </tt>
<a name="L114"></a><tt class="py-lineno">114</tt>  <tt class="py-line">            <tt class="py-name">x_todo</tt><tt class="py-op">.</tt><tt class="py-name">append</tt><tt class="py-op">(</tt><tt class="py-name">x0</tt> <tt class="py-op">+</tt> <tt class="py-name">eps</tt><tt class="py-op">)</tt> </tt>
<a name="L115"></a><tt class="py-lineno">115</tt>  <tt class="py-line"> </tt>
<a name="L116"></a><tt class="py-lineno">116</tt>  <tt class="py-line">        <tt class="py-comment"># Break that list apart into work for each node</tt> </tt>
<a name="L117"></a><tt class="py-lineno">117</tt>  <tt class="py-line">        <tt class="py-name">x_by_node</tt> <tt class="py-op">=</tt> <tt class="py-op">[</tt><tt class="py-op">]</tt> </tt>
<a name="L118"></a><tt class="py-lineno">118</tt>  <tt class="py-line">        <tt class="py-keyword">for</tt> <tt class="py-name">node_ii</tt> <tt class="py-keyword">in</tt> <tt class="py-name">range</tt><tt class="py-op">(</tt><tt id="link-33" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-33', 'num_procs', 'link-3');">num_procs</a></tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L119"></a><tt class="py-lineno">119</tt>  <tt class="py-line">            <tt class="py-name">x_by_node</tt><tt class="py-op">.</tt><tt class="py-name">append</tt><tt class="py-op">(</tt><tt class="py-name">x_todo</tt><tt class="py-op">[</tt><tt class="py-name">node_ii</tt><tt class="py-op">:</tt><tt class="py-op">:</tt><tt id="link-34" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-34', 'num_procs', 'link-3');">num_procs</a></tt><tt class="py-op">]</tt><tt class="py-op">)</tt> </tt>
<a name="L120"></a><tt class="py-lineno">120</tt>  <tt class="py-line"> </tt>
<a name="L121"></a><tt class="py-lineno">121</tt>  <tt class="py-line">        <tt class="py-comment"># Master sends orders to all other nodes.</tt> </tt>
<a name="L122"></a><tt class="py-lineno">122</tt>  <tt class="py-line">        <tt class="py-name">command</tt> <tt class="py-op">=</tt> <tt class="py-string">'eval_func_over_list(func_str, x_todo, *grad_args)'</tt> </tt>
<a name="L123"></a><tt class="py-lineno">123</tt>  <tt class="py-line">        <tt class="py-keyword">for</tt> <tt class="py-name">node_ii</tt> <tt class="py-keyword">in</tt> <tt class="py-name">range</tt><tt class="py-op">(</tt><tt class="py-number">1</tt><tt class="py-op">,</tt> <tt id="link-35" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-35', 'num_procs', 'link-3');">num_procs</a></tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L124"></a><tt class="py-lineno">124</tt>  <tt class="py-line">            <tt class="py-name">x_this_node</tt> <tt class="py-op">=</tt> <tt class="py-name">x_by_node</tt><tt class="py-op">[</tt><tt class="py-name">node_ii</tt><tt class="py-op">]</tt> </tt>
<a name="L125"></a><tt class="py-lineno">125</tt>  <tt class="py-line">            <tt class="py-name">arguments</tt> <tt class="py-op">=</tt> <tt class="py-op">{</tt><tt class="py-string">'func_str'</tt><tt class="py-op">:</tt> <tt class="py-name">func_str</tt><tt class="py-op">,</tt> </tt>
<a name="L126"></a><tt class="py-lineno">126</tt>  <tt class="py-line">                         <tt class="py-string">'x_todo'</tt><tt class="py-op">:</tt> <tt class="py-name">x_this_node</tt><tt class="py-op">}</tt> </tt>
<a name="L127"></a><tt class="py-lineno">127</tt>  <tt class="py-line">            <tt class="py-keyword">if</tt> <tt class="py-name">send_grad_args</tt><tt class="py-op">:</tt> </tt>
<a name="L128"></a><tt class="py-lineno">128</tt>  <tt class="py-line">                <tt class="py-name">arguments</tt><tt class="py-op">[</tt><tt class="py-string">'grad_args'</tt><tt class="py-op">]</tt> <tt class="py-op">=</tt> <tt class="py-name">grad_args</tt> </tt>
<a name="L129"></a><tt class="py-lineno">129</tt>  <tt class="py-line">            <tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">send</tt><tt class="py-op">(</tt><tt class="py-op">(</tt><tt class="py-name">command</tt><tt class="py-op">,</tt> <tt class="py-name">arguments</tt><tt class="py-op">)</tt><tt class="py-op">,</tt> <tt class="py-name">node_ii</tt><tt class="py-op">)</tt> </tt>
<a name="L130"></a><tt class="py-lineno">130</tt>  <tt class="py-line">        <tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">stdout</tt><tt class="py-op">.</tt><tt class="py-name">flush</tt><tt class="py-op">(</tt><tt class="py-op">)</tt> </tt>
<a name="L131"></a><tt class="py-lineno">131</tt>  <tt class="py-line"> </tt>
<a name="L132"></a><tt class="py-lineno">132</tt>  <tt class="py-line">        <tt class="py-comment"># This will hold the function evaluations done by each node.</tt> </tt>
<a name="L133"></a><tt class="py-lineno">133</tt>  <tt class="py-line">        <tt class="py-name">vals_by_node</tt> <tt class="py-op">=</tt> <tt class="py-op">[</tt><tt class="py-op">]</tt> </tt>
<a name="L134"></a><tt class="py-lineno">134</tt>  <tt class="py-line">        <tt class="py-comment"># The master node does its share of the work now</tt> </tt>
<a name="L135"></a><tt class="py-lineno">135</tt>  <tt class="py-line">        <tt class="py-name">vals_by_node</tt><tt class="py-op">.</tt><tt class="py-name">append</tt><tt class="py-op">(</tt><tt id="link-36" class="py-name" targets="Function dadi.RunInParallel.eval_func_over_list()=dadi.RunInParallel-module.html#eval_func_over_list"><a title="dadi.RunInParallel.eval_func_over_list" class="py-name" href="#" onclick="return doclink('link-36', 'eval_func_over_list', 'link-36');">eval_func_over_list</a></tt><tt class="py-op">(</tt><tt class="py-name">func_str</tt><tt class="py-op">,</tt> <tt class="py-name">x_by_node</tt><tt class="py-op">[</tt><tt class="py-number">0</tt><tt class="py-op">]</tt><tt class="py-op">,</tt>  </tt>
<a name="L136"></a><tt class="py-lineno">136</tt>  <tt class="py-line">                                                <tt class="py-op">*</tt><tt class="py-name">grad_args</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
<a name="L137"></a><tt class="py-lineno">137</tt>  <tt class="py-line">        <tt class="py-comment"># Now receive the work done by each of the other nodes</tt> </tt>
<a name="L138"></a><tt class="py-lineno">138</tt>  <tt class="py-line">        <tt class="py-keyword">for</tt> <tt class="py-name">node_ii</tt> <tt class="py-keyword">in</tt> <tt class="py-name">range</tt><tt class="py-op">(</tt><tt class="py-number">1</tt><tt class="py-op">,</tt> <tt id="link-37" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-37', 'num_procs', 'link-3');">num_procs</a></tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L139"></a><tt class="py-lineno">139</tt>  <tt class="py-line">            <tt class="py-name">vals_by_node</tt><tt class="py-op">.</tt><tt class="py-name">append</tt><tt class="py-op">(</tt><tt class="py-name">pypar</tt><tt class="py-op">.</tt><tt class="py-name">receive</tt><tt class="py-op">(</tt><tt class="py-name">node_ii</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
<a name="L140"></a><tt class="py-lineno">140</tt>  <tt class="py-line"> </tt>
<a name="L141"></a><tt class="py-lineno">141</tt>  <tt class="py-line">        <tt class="py-comment"># Reform the function value list that's broken apart by node.</tt> </tt>
<a name="L142"></a><tt class="py-lineno">142</tt>  <tt class="py-line">        <tt class="py-name">func_evals</tt> <tt class="py-op">=</tt> <tt class="py-name">numpy</tt><tt class="py-op">.</tt><tt class="py-name">zeros</tt><tt class="py-op">(</tt><tt class="py-name">len</tt><tt class="py-op">(</tt><tt class="py-name">x0</tt><tt class="py-op">)</tt><tt class="py-op">+</tt><tt class="py-number">1</tt><tt class="py-op">,</tt> <tt class="py-name">numpy</tt><tt class="py-op">.</tt><tt class="py-name">float_</tt><tt class="py-op">)</tt> </tt>
<a name="L143"></a><tt class="py-lineno">143</tt>  <tt class="py-line">        <tt class="py-keyword">for</tt> <tt class="py-name">node_ii</tt><tt class="py-op">,</tt><tt class="py-name">vals_this_node</tt> <tt class="py-keyword">in</tt> <tt class="py-name">enumerate</tt><tt class="py-op">(</tt><tt class="py-name">vals_by_node</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L144"></a><tt class="py-lineno">144</tt>  <tt class="py-line">            <tt class="py-name">func_evals</tt><tt class="py-op">[</tt><tt class="py-name">node_ii</tt><tt class="py-op">:</tt><tt class="py-op">:</tt><tt id="link-38" class="py-name"><a title="dadi.RunInParallel.num_procs" class="py-name" href="#" onclick="return doclink('link-38', 'num_procs', 'link-3');">num_procs</a></tt><tt class="py-op">]</tt> <tt class="py-op">=</tt> <tt class="py-name">vals_this_node</tt> </tt>
<a name="L145"></a><tt class="py-lineno">145</tt>  <tt class="py-line"> </tt>
<a name="L146"></a><tt class="py-lineno">146</tt>  <tt class="py-line">        <tt class="py-comment"># Now calculate the gradient</tt> </tt>
<a name="L147"></a><tt class="py-lineno">147</tt>  <tt class="py-line">        <tt class="py-name">grad</tt> <tt class="py-op">=</tt> <tt class="py-name">numpy</tt><tt class="py-op">.</tt><tt class="py-name">zeros</tt><tt class="py-op">(</tt><tt class="py-name">len</tt><tt class="py-op">(</tt><tt class="py-name">x0</tt><tt class="py-op">)</tt><tt class="py-op">,</tt> <tt class="py-name">numpy</tt><tt class="py-op">.</tt><tt class="py-name">float_</tt><tt class="py-op">)</tt> </tt>
<a name="L148"></a><tt class="py-lineno">148</tt>  <tt class="py-line">        <tt class="py-name">f0</tt> <tt class="py-op">=</tt> <tt class="py-name">func_evals</tt><tt class="py-op">[</tt><tt class="py-number">0</tt><tt class="py-op">]</tt> </tt>
<a name="L149"></a><tt class="py-lineno">149</tt>  <tt class="py-line">        <tt class="py-keyword">for</tt> <tt class="py-name">ii</tt><tt class="py-op">,</tt><tt class="py-name">func_val</tt> <tt class="py-keyword">in</tt> <tt class="py-name">enumerate</tt><tt class="py-op">(</tt><tt class="py-name">func_evals</tt><tt class="py-op">[</tt><tt class="py-number">1</tt><tt class="py-op">:</tt><tt class="py-op">]</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
<a name="L150"></a><tt class="py-lineno">150</tt>  <tt class="py-line">            <tt class="py-name">grad</tt><tt class="py-op">[</tt><tt class="py-name">ii</tt><tt class="py-op">]</tt> <tt class="py-op">=</tt> <tt class="py-op">(</tt><tt class="py-name">func_val</tt> <tt class="py-op">-</tt> <tt class="py-name">f0</tt><tt class="py-op">)</tt><tt class="py-op">/</tt><tt class="py-name">epsilon</tt> </tt>
<a name="L151"></a><tt class="py-lineno">151</tt>  <tt class="py-line"> </tt>
<a name="L152"></a><tt class="py-lineno">152</tt>  <tt class="py-line">        <tt class="py-keyword">return</tt> <tt class="py-name">grad</tt> </tt>
</div><a name="L153"></a><tt class="py-lineno">153</tt>  <tt class="py-line">    <tt class="py-keyword">return</tt> <tt class="py-name">grad_func</tt> </tt>
</div><a name="L154"></a><tt class="py-lineno">154</tt>  <tt class="py-line"> </tt><script type="text/javascript">
<!--
expandto(location.href);
// -->
</script>
</pre>
<br />
<!-- ==================== NAVIGATION BAR ==================== -->
<table class="navbar" border="0" width="100%" cellpadding="0"
       bgcolor="#a0c0ff" cellspacing="0">
  <tr valign="middle">

  <!-- Tree link -->
      <th>&nbsp;&nbsp;&nbsp;<a
        href="module-tree.html">Trees</a>&nbsp;&nbsp;&nbsp;</th>

  <!-- Index link -->
      <th>&nbsp;&nbsp;&nbsp;<a
        href="identifier-index.html">Indices</a>&nbsp;&nbsp;&nbsp;</th>

  <!-- Help link -->
      <th>&nbsp;&nbsp;&nbsp;<a
        href="help.html">Help</a>&nbsp;&nbsp;&nbsp;</th>

  <!-- Project homepage -->
      <th class="navbar" align="right" width="100%">
        <table border="0" cellpadding="0" cellspacing="0">
          <tr><th class="navbar" align="center"
            ><a class="navbar" target="_top" href="http://dadi.googlecode.com">dadi</a></th>
          </tr></table></th>
  </tr>
</table>
<table border="0" cellpadding="0" cellspacing="0" width="100%%">
  <tr>
    <td align="left" class="footer">
    <a href="epydoc-log.html">Generated by Epydoc
    3.0.1 on Fri Dec 14 15:16:35 2012</a>
    </td>
    <td align="right" class="footer">
      <a target="mainFrame" href="http://epydoc.sourceforge.net"
        >http://epydoc.sourceforge.net</a>
    </td>
  </tr>
</table>

<script type="text/javascript">
  <!--
  // Private objects are initially displayed (because if
  // javascript is turned off then we want them to be
  // visible); but by default, we want to hide them.  So hide
  // them unless we have a cookie that says to show them.
  checkCookie();
  // -->
</script>
</body>
</html>
